#!/usr/bin/env python3

# 2023 TAE Technologies

import numpy as np

# fmt: off
ref_density = np.array([1.27936406e+14, 2.23582812e+14, 2.55397239e+14, 2.55665600e+14,
                        2.55816698e+14, 2.55809992e+14, 2.55805185e+14, 2.55741476e+14,
                        2.55962733e+14, 2.56255357e+14, 2.56154079e+14, 2.55928556e+14,
                        2.55910663e+14, 2.56047545e+14, 2.56254851e+14, 2.56272722e+14,
                        2.56104357e+14, 2.56081559e+14, 2.56062218e+14, 2.56090169e+14,
                        2.56208624e+14, 2.56382560e+14, 2.56330895e+14, 2.55960535e+14,
                        2.56043442e+14, 2.56392987e+14, 2.56504533e+14, 2.56470312e+14,
                        2.56362683e+14, 2.56231920e+14, 2.56221126e+14, 2.56378751e+14,
                        2.56627372e+14, 2.56934554e+14, 2.56942385e+14, 2.56634523e+14,
                        2.56374255e+14, 2.56166775e+14, 2.56077406e+14, 2.56293743e+14,
                        2.56524129e+14, 2.56447966e+14, 2.56212174e+14, 2.56158246e+14,
                        2.56314253e+14, 2.56602675e+14, 2.56689737e+14, 2.56464033e+14,
                        2.56516448e+14, 2.56548092e+14, 2.56294609e+14, 2.56148212e+14,
                        2.56107543e+14, 2.56169576e+14, 2.56367742e+14, 2.56302966e+14,
                        2.55973839e+14, 2.55930017e+14, 2.56113703e+14, 2.56156741e+14,
                        2.56217970e+14, 2.56563645e+14, 2.56624919e+14, 2.56313759e+14,
                        2.56281864e+14, 2.56301642e+14, 2.56172954e+14, 2.56191364e+14,
                        2.56408830e+14, 2.56670084e+14, 2.56884275e+14, 2.57049327e+14,
                        2.56963937e+14, 2.56567686e+14, 2.56436000e+14, 2.56618943e+14,
                        2.56696293e+14, 2.56674448e+14, 2.56499604e+14, 2.56543063e+14,
                        2.56777472e+14, 2.56717884e+14, 2.56504733e+14, 2.56297269e+14,
                        2.56315988e+14, 2.56440995e+14, 2.56325109e+14, 2.56236960e+14,
                        2.56279548e+14, 2.56327503e+14, 2.56397343e+14, 2.56354107e+14,
                        2.56173972e+14, 2.56301766e+14, 2.56768978e+14, 2.56833877e+14,
                        2.56469382e+14, 2.56293263e+14, 2.56433467e+14, 2.56586998e+14,
                        2.56774870e+14, 2.56785814e+14, 2.56422956e+14, 2.56250689e+14,
                        2.56498806e+14, 2.56887911e+14, 2.56873314e+14, 2.56739125e+14,
                        2.56795711e+14, 2.56592411e+14, 2.56568317e+14, 2.56844348e+14,
                        2.56803216e+14, 2.56460213e+14, 2.56419149e+14, 2.56603917e+14,
                        2.56498541e+14, 2.56253989e+14, 2.56316482e+14, 2.56825819e+14,
                        2.57048376e+14, 2.56916510e+14, 2.56689528e+14, 2.56366255e+14,
                        2.56307403e+14, 2.56292497e+14, 2.56227982e+14, 2.24009867e+14,
                        1.27761381e+14])
# fmt: on

density_data = np.load("ion_density_case_1.npy")
print(repr(density_data))
assert np.allclose(density_data, ref_density)
